package org.ls.petclinic.untils;

import com.alibaba.druid.pool.DruidDataSource;
import com.alibaba.druid.pool.DruidDataSourceFactory;

import java.io.IOException;
import java.io.InputStream;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.Properties;

public class JdbcUtils {
    private static DruidDataSource dataSource;
    private static ThreadLocal<Connection> conns=new ThreadLocal<Connection>();

    // 静态代码块初始化加载驱动
    static {
        Properties properties = new Properties();
        // 通过类加载器来获得流
        InputStream inputStream = JdbcUtils.class.getResourceAsStream("/druid.properties");
        try {
            // 通过流读取配置文件中的内容到集合中
            properties.load(inputStream);
            // 通过Druid工厂加载文件注册驱动,初始化池子
            dataSource = (DruidDataSource) DruidDataSourceFactory.createDataSource(properties);

        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        } catch (Exception e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }

    }


    /**
    * 获取数据库连接池中的连接
    * @return
     */
    public static Connection getConnection(){
       Connection connection=conns.get();
       if (connection==null){
           try {
               connection= dataSource.getConnection();//从数据库连接池中获取连接
               conns.set(connection);//保存到ThreadLocal对象中,供后面的jdbc使用
               connection.setAutoCommit(false);//设置为自动管理事务
           } catch (SQLException e) {
               e.printStackTrace();
           }
       }
       return connection;
    }

    /**
     * 提交事务,并关闭释放资源
     */
    public static void commitAndClose(){
        Connection connection = conns.get();
        if (connection!=null){//如果不等于null,说明之前使用过连接,操作过数据库
            try {
                connection.commit();//提交事务

            } catch (SQLException e) {
                e.printStackTrace();
            }finally {
                try {
                    connection.close();//关闭连接
                } catch (SQLException e) {
                    e.printStackTrace();
                }
            }
            conns.remove();//一定要执行remove操作,否则就会出错(因为Tomcat服务器底层使用了线程池技术)
        }
    }
    /**
     * 回滚事务,并关闭释放资源
     */
    public static void rollbackAndClose(){
        Connection connection = conns.get();
        if (connection!=null){//如果不等于null,说明之前使用过连接,操作过数据库
            try {
                connection.rollback();//提交事务

            } catch (SQLException e) {
                e.printStackTrace();
            }finally {
                try {
                    connection.close();//关闭连接
                } catch (SQLException e) {
                    e.printStackTrace();
                }
            }
            conns.remove();//一定要执行remove操作,否则就会出错(因为Tomcat服务器底层使用了线程池技术)
        }
    }

    /**
     * 提交事务,并关闭释放资源
     */
    public static void commit(){
        Connection connection = conns.get();
        if (connection!=null){//如果不等于null,说明之前使用过连接,操作过数据库
            try {
                connection.commit();//提交事务

            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
    }
    /**
     * 关闭释放资源
     */
    public static void close(){
        Connection connection = conns.get();
        if (connection!=null){//如果不等于null,说明之前使用过连接,操作过数据库
            try {
                connection.close();//提交事务

            } catch (SQLException e) {
                e.printStackTrace();
            }
            conns.remove();//一定要执行remove操作,否则就会出错(因为Tomcat服务器底层使用了线程池技术)
        }
    }
}
